import os

from openai import OpenAI


def call_openai_model(prompt):
    client = OpenAI(
        api_key=os.getenv('OPENAI_KEY'),
        base_url=os.getenv('BASE_URL'),
    )
    response = client.chat.completions.create(
        model="gpt-3.5-turbo",
        messages=[{"role": "user", "content": prompt}],
    )
    return response.choices[0].message.content